from django.core.management.base import BaseCommand
from django.contrib.auth.models import User

class Command(BaseCommand):
    help = '创建测试用户账号'

    def handle(self, *args, **options):
        # 创建超级用户
        if not User.objects.filter(username='admin').exists():
            User.objects.create_superuser(
                username='admin',
                email='admin@example.com',
                password='admin123'
            )
            self.stdout.write('✅ 超级用户创建成功: admin / admin123')
        else:
            self.stdout.write('⚠️ 超级用户 admin 已存在')

        # 创建普通测试用户
        if not User.objects.filter(username='test').exists():
            User.objects.create_user(
                username='test',
                email='test@example.com',
                password='test123'
            )
            self.stdout.write('✅ 测试用户创建成功: test / test123')
        else:
            self.stdout.write('⚠️ 测试用户 test 已存在')

        self.stdout.write('\n🎉 用户创建完成！您现在可以使用以下账号测试前端登录：')
        self.stdout.write('   管理员: admin / admin123')
        self.stdout.write('   普通用户: test / test123')